from django.db import models
from django.contrib.auth.models import AbstractUser
from itsdangerous import TimedJSONWebSignatureSerializer as TJSWSSerializer, BadData
from django.conf import settings

from utils.models import BaseModel

# 用戶
class User(AbstractUser):
    mobile = models.CharField(max_length=11,unique=True, verbose_name="手機號")
    email_active = models.BooleanField(default=False,verbose_name="郵箱激活狀態")
    default_address = models.ForeignKey('Address', related_name='users', null=True, blank=True,on_delete=models.SET_NULL, verbose_name='默认地址')

    class Meta:
        db_table = "tb_users"


    def __str__(self):
        return self.username

    # 生成驗證的鏈接
    def generate_verify_url(self):
        # 生成數據
        token = {"user_id":self.id, "email":self.email}

        # 加密數據
        serializer = TJSWSSerializer(secret_key=settings.SECRET_KEY,expires_in=300)
        token = serializer.dumps(token)

        # 拼接到url地址中
        verify_url = "http://www.meiduo.site:8080/success_verify_email.html?token=%s"%token.decode()

        return verify_url

    # 驗證token
    @staticmethod
    def check_verify_email_token(token):
        # 解密數據
        serializer = TJSWSSerializer(secret_key=settings.SECRET_KEY,expires_in=300)
        try:
            dict_token = serializer.loads(token)
        except BadData:
            return None
        else:
            # 取出id,email
            user_id = dict_token.get("user_id")
            email = dict_token.get("email")
        try:
            user = User.objects.get(id=user_id,email=email)
        except User.DoesNotExist:
            user = None
        else:
            return user


# 用戶地址
class Address(BaseModel):
    """
    用户地址
    """
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='addresses', verbose_name='用户')
    title = models.CharField(max_length=20, verbose_name='地址名称')
    receiver = models.CharField(max_length=20, verbose_name='收货人')
    province = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='province_addresses', verbose_name='省')
    city = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='city_addresses', verbose_name='市')
    district = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='district_addresses', verbose_name='区')
    place = models.CharField(max_length=50, verbose_name='地址')
    mobile = models.CharField(max_length=11, verbose_name='手机')
    tel = models.CharField(max_length=20, null=True, blank=True, default='', verbose_name='固定电话')
    email = models.CharField(max_length=30, null=True, blank=True, default='', verbose_name='电子邮箱')
    is_deleted = models.BooleanField(default=False, verbose_name='逻辑删除')

    class Meta:
        db_table = 'tb_address'
        verbose_name = '用户地址'
        verbose_name_plural = verbose_name
        ordering = ['-update_time']


